import torch
print(torch.cuda.is_available())
print(torch.cuda_version)
print(torch.cuda.get_device_name(0))  # 显示GPU型号（如"NVIDIA RTX 3090"）

#ckpt = torch.load("runs/detect/train21/weights/best.pt")
#ckpt["epoch"] = 348  # 将 -1 改为已完成的 400 轮（从 401 开始续训）
#torch.save(ckpt, "runs/detect/train21/weights/best_modified.pt")


#import paddle
#paddle.utils.run_check()